"""
author：fc
date：  2021/10/18
"""
#
# 天善智能人工神经网络示例，2016年的视频，当时很前沿了吧
#
import pandas as pd
import numpy as np
from keras.models import Sequential
from keras.layers.core import Dense,Activation
data=pd.read_excel("../files/数据建模/lesson.xlsx",engine="openpyxl")
x=data.iloc[:,1:5]
y=data.iloc[:,5:6]
for column in x: # 这种还是没有循环方便,我花时间在多次一举
    x[column][x[column] =="是"]=1
    x[column][x[column] =="多"]=1
    x[column][x[column]!=1]=-1
y["销量"][y["销量"]=="高"]=1
y["销量"][y["销量"]!=1]=-1
x=x.value.tolist()
y=y.value.tolist()
model=Sequential()
# 输入层
model.add(Dense(10,input_dim=4))
model.add(Activation("relu"))
# 输出层
model.add(Dense(1,1))
model.add(Activation("sigmoid"))
# 模型编译
model.compile(loss="binary_crossentropy",optimizer="adam") # adam是求解方法
# 训练
model.fit(x,y,nb_epoch=1000,batch_size=100)
# 预测分类
test_data=np.array([[1,-1,-1,1]]) #数据预测
model.predict_classes(test_data).reshape().reshape(len(test_data))